// Copyright 2021 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

// Generated by the Codegen C++ plugin.
// If you make any local changes, they will be lost.
// source: google/cloud/bigquery/storage/v1/storage.proto

#include "google/cloud/bigquery/bigquery_read_client.h"
#include <memory>

namespace google {
namespace cloud {
namespace bigquery {
inline namespace GOOGLE_CLOUD_CPP_GENERATED_NS {

BigQueryReadClient::BigQueryReadClient(
    std::shared_ptr<BigQueryReadConnection> connection)
    : connection_(std::move(connection)) {}
BigQueryReadClient::~BigQueryReadClient() = default;

StatusOr<google::cloud::bigquery::storage::v1::ReadSession>
BigQueryReadClient::CreateReadSession(
    std::string const& parent,
    google::cloud::bigquery::storage::v1::ReadSession const& read_session,
    std::int32_t max_stream_count) {
  google::cloud::bigquery::storage::v1::CreateReadSessionRequest request;
  request.set_parent(parent);
  *request.mutable_read_session() = read_session;
  request.set_max_stream_count(max_stream_count);
  return connection_->CreateReadSession(request);
}

StreamRange<google::cloud::bigquery::storage::v1::ReadRowsResponse>
BigQueryReadClient::ReadRows(std::string const& read_stream,
                             std::int64_t offset) {
  google::cloud::bigquery::storage::v1::ReadRowsRequest request;
  request.set_read_stream(read_stream);
  request.set_offset(offset);
  return connection_->ReadRows(request);
}

StatusOr<google::cloud::bigquery::storage::v1::ReadSession>
BigQueryReadClient::CreateReadSession(
    google::cloud::bigquery::storage::v1::CreateReadSessionRequest const&
        request) {
  return connection_->CreateReadSession(request);
}

StreamRange<google::cloud::bigquery::storage::v1::ReadRowsResponse>
BigQueryReadClient::ReadRows(
    google::cloud::bigquery::storage::v1::ReadRowsRequest const& request) {
  return connection_->ReadRows(request);
}

StatusOr<google::cloud::bigquery::storage::v1::SplitReadStreamResponse>
BigQueryReadClient::SplitReadStream(
    google::cloud::bigquery::storage::v1::SplitReadStreamRequest const&
        request) {
  return connection_->SplitReadStream(request);
}

}  // namespace GOOGLE_CLOUD_CPP_GENERATED_NS
}  // namespace bigquery
}  // namespace cloud
}  // namespace google
